package com.servlet;

import org.apache.commons.fileupload.FileItem;
import org.apache.commons.fileupload.FileItemFactory;
import org.apache.commons.fileupload.FileUploadException;
import org.apache.commons.fileupload.disk.DiskFileItemFactory;
import org.apache.commons.fileupload.servlet.ServletFileUpload;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.File;
import java.io.IOException;
import java.util.List;

@WebServlet("/fileupload.do")
public class FileUploadServlet extends HttpServlet {
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {

    }

    @Override
    protected void doPost(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //1.判断上传的数据是否为多段表单数据（multipart-formdata）
        if (ServletFileUpload.isMultipartContent(req)) {
            //创建FileItemFactory 工厂实现类
            FileItemFactory fileItemFactory = new DiskFileItemFactory();
            //创建用于解析上传数据的工具类 ServletFileUpload
            ServletFileUpload servletFileUpload = new ServletFileUpload(fileItemFactory);
            try {
                //解析上传的数据 得到每个表单项FileItem
                List<FileItem> list = servletFileUpload.parseRequest(req);
                //循环判断：每一个表单项是普通文本类型，还是文件类型
                for (int i = 0; i < list.size(); i++) {
                    FileItem fileItem = list.get(i);
                    if (fileItem.isFormField()) {
                        //普通表单项
                        System.out.println("属性名：" + fileItem.getFieldName());
                        System.out.println("属性值：" + fileItem.getString("UTF-8"));
                    } else {
                        //是文件类型
                        System.out.println("属性名：" + fileItem.getFieldName());
                        System.out.println("文件名：" + fileItem.getName());
                        fileItem.write(new File("D:\\imgUpload\\" + fileItem.getName()));
                    }
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }
}
